import mysql.connector
import json

# 创建连接
cnx = mysql.connector.connect(user='root', password='123456',
                              host='localhost', database='bus_view')

# 创建游标对象
cursor = cnx.cursor()

# 执行查询
query = ("SELECT station_name,longitude,latitude FROM station_details")
cursor.execute(query)

# 初始化一个空字典来存储结果
result_dict = {}

# 遍历查询结果
for (station_name, longitude, latitude) in cursor:
    # 将经度和纬度添加到字典中
    # 注意，这里我们使用 float() 函数将 Decimal 类型的数据转换为 float 类型
    result_dict[station_name] = [float(longitude)+0.006681, float(latitude)+0.005708]

# 将字典转换为JSON字符串
json_str = json.dumps(result_dict, indent=4,ensure_ascii=False)

# 将JSON字符串写入文件
with open('../output.json', 'w') as f:
    f.write(json_str)

# 关闭游标和连接
cursor.close()
cnx.close()